# 小明的桌面 乱了, 有很多类型的文件

# 请你写一个脚本  帮助小明将不同类型的文件   移动到 对应的文件夹里

# 图片(png, bpm,gif,jpg..)放到 pic 文件夹
# 文本文件(txt,doc,docx)    放到 doc文件夹
# 音频文件(mp3,) 放到  voice 文件夹
# 视频文件(mp4,avi...) 放到 video 文件夹

# 以上文件夹 放在 D盘的 auto 文件夹中
# D:/auto/pic

from os import path
import os
import shutil

map = ['png','bpm','gif','jpg']
txt = ['txt','doc','docx']
mp3 = 'mp3'
video = ['mp4','avi']


def judge(list_files):
    global list_format
    list_format = []
    for i in list_files:
        file_format = i[-4:]
        if '.' in file_format:
            list1 = file_format.split()
            list1.remove('.')
            list_format.append(file_format)


# def make_move_dir(list_files):
#     os.mkdir('d:/auto')
#     judge(list_files)
#     for f in list_files:
#         for i in list_format:
#             if i in map:
#                 os.mkdir('pic')
#                 shutil.move(f,'d:auto')
#
#             elif i in txt:
#                 os.mkdir('doc')
#                 shutil.move(f, 'd:auto')
#
#             elif i in mp3:
#                 os.mkdir('voice')
#                 shutil.move(f, 'd:auto')
#
#             elif i in video:
#                 os.mkdir('video')
#                 shutil.move(f, 'd:auto')
#
list_files = ['哈哈.txt','嗯嗯.png','嗯哼.mp3','哦豁.mp4']
judge(list_files)
print(list_format)








